package com.bdqn.bookshop.utils;


import com.bdqn.bookshop.entity.Classify;

import java.util.ArrayList;
import java.util.List;
/**
 * @author : 罗若然[LUO_RUO_RAN]
 * @date : 2020-07-25 14:17
 **/
public class ClassifyNodeBuilder {

    /**
     * 创建节点层级关系
     * @param classify     节点对象集合
     * @return
     */
    public static List<Classify> build(List<Classify> classify){
        List<Classify> classifies= new ArrayList<Classify>();
        //循环遍历节点集合
        for (Classify n1 : classify) {
            //如果当前节点为根节点，则将当前节点添加到节点数组中
            if(n1.getParentId()==0) {
                classifies.add(n1);
            }
            //如果当前子节点对应的节点相等，则添加到子节点集合中
            for (Classify n2 : classify) {
                if(n1.getId()==n2.getParentId()&& n2.getParentId()>0) {
                    n1.getClassifyList().add(n2);
                }
            }
        }

        return classifies;
    }
}
